import pandas as pd
import pickle

file_path = "../logs/head_importance/opt66b/1shot_piqa.pkl"
with open(file_path, 'rb') as f:
    tensor_data = pickle.load(f)

print(tensor_data)  # 显示 Tensor 内容
print(tensor_data.shape)  # 显示 Tensor 形状（如果有 shape 属性）